package com.heiye.blog.ai.enums;

import lombok.AllArgsConstructor;
import lombok.Getter;

/**
 * @author: heiye
 * @date: 2025/11/07 下午5:23
 * @version: v1.0.0
 * @description: 文本模型枚举
 */
@Getter
@AllArgsConstructor
public enum TextModelTypeEnum {
    // deepseek
    // 深度思考
    DEEPSEEK_REASONER("deepseek-reasoner"),
    // 不深度思考
    DEEPSEEK_CHAT("deepseek-chat"),

    // 基于阿里云百炼模型的文本模型
    DASH_SCOPE_DEEPSEEK_V3("deepseek-v3"),
    DASH_SCOPE_DEEPSEEK_R1("deepseek-r1"),
    DASH_SCOPE_DEEPSEEK_V3_2_EXP("deepseek-v3.2-exp"),

    // 谷歌文本模型
    GOOGLE_GEMINI_2_5_FLASH("google/gemini-2.5-flash"),
    GOOGLE_GEMINI_2_5_PRO("google/gemini-2.5-pro"),
    ;

    private final String type;

    public static TextModelTypeEnum getTextModelType(String type) {
        for (TextModelTypeEnum value : TextModelTypeEnum.values()) {
            if (value.getType().equals(type)) {
                return value;
            }
        }
        return null;
    }
}
